package com.heima.utils.common;

import com.heima.model.wemedia.pojos.WmUser;

/**
 * @author John
 * @date 2022/12/10 0010
 * @apiNote
 */
public class WmThreadLocalUtil {
    private final static ThreadLocal<WmUser> WM_USER_THREAD_LOCAL = new ThreadLocal<>();

    /**
     * 添加用户
     * @param wmUser
     */
    public static void setUser(WmUser wmUser) {
        WM_USER_THREAD_LOCAL.set(wmUser);
    }

    /**
     * 获取用户
     * @param
     * @return
     */
    public static WmUser getUser() {
        return WM_USER_THREAD_LOCAL.get();
    }

    /**
     * 清理用户
     */
    public static void remove() {
        WM_USER_THREAD_LOCAL.remove();
    }


}
